input in array in python